package cn.redis.util;

import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * 基于 StringRedisTemplate 的 Redis工具类。用于避免显式指定泛型或者不指定导致的代码不好维护的问题。
 * 如果需要存储对象信息，请手动将其转换成json字符串格式进行存储。
 *
 * @author yexi
 */

@Component
@Slf4j
public class RedisUtil {

    @Autowired
    private StringRedisTemplate redisTemplate;

    /**
     * string类型缓存获取
     *
     * @param key 键
     * @return 值
     */
    public String get(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * string类型缓存设置
     *
     * @param key   键
     * @param value 值
     * @return true成功 false失败
     */
    public boolean set(String key, String value) {
        try {
            redisTemplate.opsForValue().set(key, value);
            return true;
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return false;
        }
    }

    /**
     * 普通缓存放入并设置过期时间
     *
     * @param key   键
     * @param value 值
     * @param time  时间(秒) time要大于0 如果time小于等于0 将设置无限期
     * @return true成功 false 失败
     */
    public boolean set(String key, String value, long time) {
        try {
            if (time > 0) {
                redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
                return true;
            } else {
                return this.set(key, value);
            }
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return false;
        }
    }

    /**
     * zset类型缓存设置
     *
     * @param key   键
     * @param value 对象的json格式
     * @param score 排序标识
     * @return true成功 false 失败
     */
    public boolean zadd(String key, String value, double score) {
        try {
            ZSetOperations<String, String> zSetOperations = redisTemplate.opsForZSet();
            zSetOperations.add(key, value, score);
            return true;
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return false;
        }
    }

    /**
     * 根据 score 范围返回指定 key 的内容(zset数据格式)，按升序返回，返回的列表通过 clazz 转为指定对象类型列表
     *
     * @param key   键
     * @param min   最小的score
     * @param max   最大的score
     * @param clazz 要转换的类型
     * @return 符合条件的对象列表
     */
    public <T> List<T> zrangeByScore(String key, double min, double max, Class<T> clazz) {
        try {
            ZSetOperations<String, String> zSetOperations = redisTemplate.opsForZSet();
            Set<String> range = zSetOperations.rangeByScore(key, min, max);
            return CollectionUtils.isEmpty(range) ? new ArrayList<>() : JSON.parseArray(range.toString(), clazz);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return new ArrayList<>();
        }
    }

    /**
     * 同上，按降序返回
     */
    public <T> List<T> zrevrangeByScore(String key, double min, double max, Class<T> clazz) {
        try {
            ZSetOperations<String, String> zSetOperations = redisTemplate.opsForZSet();
            Set<String> range = zSetOperations.reverseRangeByScore(key, min, max);
            return CollectionUtils.isEmpty(range) ? new ArrayList<>() : JSON.parseArray(range.toString(), clazz);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return new ArrayList<>();
        }
    }

    /**
     * 删除redis缓存
     *
     * @param key 可以传一个值 或多个
     */
    @SuppressWarnings("unchecked")
    public void delete(String... key) {
        if (key != null && key.length > 0) {
            if (key.length == 1) {
                redisTemplate.delete(key[0]);
            } else {
                redisTemplate.delete((Collection<String>) CollectionUtils.arrayToList(key));
            }
        }
    }

    /**
     * 返回指定业务key的所有key
     *
     * @param key RedisKeys
     * @return 符合条件的所有key
     */
    public List<String> keys(RedisKeys key) {
        try {
            Set<String> keyList = redisTemplate.keys(key.join("*"));
            return CollectionUtils.isEmpty(keyList) ? new ArrayList<>() : new ArrayList<>(keyList);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            return new ArrayList<>();
        }
    }

    /**
     * 获取redis服务器时间。适用于分布式系统打算统一时间的场合。
     */
    public Long getServerTime() {
        return redisTemplate.execute((RedisCallback<Long>) connection -> connection.time());
    }

}
